Pyspark optimisation techniques